from __future__ import absolute_import

import os
from got10k.datasets import *

from siamfc import TrackerSiamFC


if __name__ == '__main__':
    # root_dir = os.path.expanduser('~/data/GOT-10k')
    # seqs = GOT10k(root_dir, subset='train', return_meta=True)
    root_dir = os.path.expanduser('/home/zs/work/learn/track/sot/dataset/OTB100')
    seqs = OTB(root_dir, version=2015, download=False)

    tracker = TrackerSiamFC()
    tracker.train_over(seqs)
